import time
from zhokao2.page import a3m
from selenium import webdriver
from selenium.webdriver.common.by import By
import unittest
from zhokao2.page2 import Pate
from zhokao2.page1 import Page
class Testcase(unittest.TestCase):
    def setUp(self):
        self.dr=webdriver.Firefox()
        self.dr.maximize_window()
    def tearDown(self):
        time.sleep(2)
        self.dr.quit()
    def test_001(self):
        c=Page(self.dr)
        a=a3m(self.dr)
        a.yx()
        c.yx1()
    def test_002(self):
        c = Pate(self.dr)
        a=a3m(self.dr)
        a.yx()
        c.yy()
if __name__ == '__main__':
    unittest.main()